package com.users.util;

import java.io.ByteArrayOutputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;

import javax.servlet.RequestDispatcher;
import javax.servlet.ServletContext;
import javax.servlet.ServletException;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpServletResponseWrapper;

public class ToIndexHtml extends HttpServlet{
	private static final long serialVersionUID = -7037447284714919866L;
	public void service(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
		request.setCharacterEncoding("UTF-8");
		response.setContentType("text/html;charset=UTF-8");
		String url = "";
		String name = "";
		ServletContext sc = getServletContext();
		String file_name = request.getParameter("file_name");// 你要访问的jsp文件名,如index，不包括扩展名
		// 则你访问这个servlet时加参数.如http://localhost/test/toHtml?file_name=index
		url = "/" + file_name;// 你要生成的页面的文件名。我的扩展名为jsf .
		String path = this.getServletContext().getRealPath("/");// 获取网站根目录
		name = path + "\\" + "index" + ".html";// 这是生成的html文件名,如index.html.文件名字与源文件名相同。扩展名为html
		// ConfConstants.CONTEXT_PATH为你的应用的上下文路径。
		RequestDispatcher rd = sc.getRequestDispatcher(url);
		final ByteArrayOutputStream os = new ByteArrayOutputStream();
		final ServletOutputStream stream = new ServletOutputStream(){
			public void write(byte[] data, int offset, int length) {
				os.write(data, offset, length);
			}
			public void write(int b) throws IOException {
				os.write(b);
			}
		};
		final PrintWriter pw = new PrintWriter(new OutputStreamWriter(os));
		HttpServletResponse rep = new HttpServletResponseWrapper(response){
			public ServletOutputStream getOutputStream() {
				return stream;
			}
			public PrintWriter getWriter() {
				return pw;
			}
		};
		rd.include(request, rep);
		pw.flush();
		FileOutputStream fos = new FileOutputStream(name); // 把jsp输出的内容写到xxx.html
		os.writeTo(fos);
		fos.close();
		PrintWriter out = response.getWriter();
		out.print("<p align=center><font size=3 color=red>页面已经成功生成！</font></p>");
		out.print("<script>location='index.html'</script>");
	}
}
